import torch
print(torch.version.cuda)  # 输出CUDA版本